from matplotlib import pyplot as plt
import numpy as np
import pandas as pd

x = np.linspace(-10, 10, 100)
y = 1 / (1 + np.exp(-x))
plt.plot(x, y)
# plt.show()

"""
2, 创建一个DataFrame(df)，用data做数据，labels做行索引
"""
data = {'animal': ['cat', 'cat', 'snake', 'dog', 'dog', 'cat', 'snake', 'cat', 'dog', 'dog'],
        'age': [2.5, 3, 0.5, 3.6, 5, 2, 4.5, 5, 7, 3],
        'visits': [1, 3, 2, 3, 2, 3, 1, 1, 2, 1],
        'priority': ['yes', 'yes', 'no', 'yes', 'no', 'no', 'no', 'yes', 'no', 'no']}

labels = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']

df = pd.DataFrame(data=data, index=labels)
print(df)
print(f"选取animal和age列的数据：\n{df[['animal', 'age']]}")
print(f"选取visits大于2的数据：\n{df[df['visits'] > 2]}")
df.loc['f', 'age'] = 1.5
print(f'将f行的age改为1.5： \n{df}')
print(f'计算visits列的数据综合： \n{df["visits"].sum()}')